from openai import OpenAI

from deepseek.DeepSeekModel import DeepSeekModel

# DeepSeek的对话工具类
class CowainAi:

    def __init__(self):
        # 出于与 OpenAI 兼容考虑，您也可以将 base_url 设置为 https://api.deepseek.com/v1 来使用，但注意，此处 v1 与模型版本无关
        self.client = OpenAI(
            api_key="123",  # 对应 OPENAI_API_KEY
            base_url="http://172.27.254.237:8080/v1"  # 对应 OPENAI_API_BASE
        )
        self.model = "Qwen3-Next-80B-A3B-Instruct"


cowainAi = CowainAi()

if __name__ == "__main__":
    response = cowainAi.client.chat.completions.create(
        model="Qwen3-Next-80B-A3B-Instruct",  # 必须与你服务中注册的模型名一致
        messages=[
            {"role": "system", "content": "你是一个乐于助人的 AI 助手。"},
            {"role": "user", "content": "你好，请介绍一下你自己。"}
        ],
        temperature=0.7,
        max_tokens=512
    )
    print(response.choices[0].message.content)